package com.bingxue.edu.management.system.model.entity;

import cn.hutool.core.util.StrUtil;
import com.bingxue.edu.common.enums.FileTypeEnum;
import com.bingxue.edu.framework.common.model.BusinessEntity;
import com.mybatisflex.annotation.Table;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.dromara.x.file.storage.core.FileInfo;


/**
 * 文件实体
 *
 * @author panbo
 * @since 2024/07/24 08:57
 */
@Data
@EqualsAndHashCode(callSuper = true)
@Table("base_file")
public class File extends BusinessEntity {

    /**
     * 名称
     */
    private String name;

    /**
     * 大小
     */
    private Long fileSize;

    /**
     * URL
     */
    private String fileUrl;

    /**
     * 扩展名
     */
    private String extension;

    /**
     * 缩略图大小
     */
    private Long thumbnailSize;

    /**
     * 缩略图URL
     */
    private String thumbnailUrl;

    /**
     * 文件类型
     */
    private FileTypeEnum fileType;

    /**
     * 存储ID
     */
    private String storageId;

    /**
     * 转换为 X-File-Storage 文件信息对象
     *
     * @param storageCode 存储编码
     * @return X-File-Storage 文件信息对象
     */
    public FileInfo toFileInfo(String storageCode) {
        FileInfo fileInfo = new FileInfo();
        fileInfo.setUrl(this.fileUrl);
        fileInfo.setSize(this.fileSize);
        fileInfo.setFilename(StrUtil.contains(this.fileUrl, StrUtil.SLASH) ? StrUtil.subAfter(this.fileUrl, StrUtil.SLASH, true) : this.fileUrl);
        fileInfo.setOriginalFilename(StrUtil.isBlank(this.extension) ? this.name : this.name + StrUtil.DOT + this.extension);
        fileInfo.setBasePath(StrUtil.EMPTY);
        fileInfo.setPath(StrUtil.subBefore(this.fileUrl, StrUtil.SLASH, true) + StrUtil.SLASH);
        fileInfo.setExt(this.extension);
        fileInfo.setPlatform(storageCode);
        fileInfo.setThUrl(this.thumbnailUrl);
        fileInfo.setThFilename(StrUtil.contains(this.thumbnailUrl, StrUtil.SLASH) ? StrUtil.subAfter(this.thumbnailUrl, StrUtil.SLASH, true) : this.thumbnailUrl);
        fileInfo.setThSize(this.thumbnailSize);
        return fileInfo;
    }
}
